#pragma once

#include <iostream>
#include <queue>
#include <pthread.h>
#include <unistd.h>
#include <cstdlib>
#include <ctime>

#define NUM 6

template<class T>
class PC
{
public:
    PC(int capacity = NUM):_capacity(capacity)
    {
        // 初始化锁和条件变量
        pthread_mutex_init(&lock,nullptr);
        pthread_cond_init(&_empty,nullptr);
        pthread_cond_init(&_full,nullptr);
    }

    void Push(const T &in)
    {
        // 先申请锁
        pthread_mutex_lock(&lock);
        // 判断是否为满
        while(full())
        {
            // 如果满了，那么就必须等待，并把锁释放出去
            pthread_cond_wait(&_full,&lock);
        }
        // 插入数据
        q.push(in);
        // 队列中已经有数据了，唤醒正在等待的消费者
        if(q.size() >= _capacity/2)
        {
            pthread_cond_signal(&_empty);
            std::cout << "消费者快来消费吧 ";
        }
        // 释放锁
        pthread_mutex_unlock(&lock);
    }

    void Pop(T &out)
    {
        // 先申请锁
        pthread_mutex_lock(&lock);
        // 判断队列是否为空
        while(empty())
        {
            // 如果为空，那就消费者等待，并释放锁
            pthread_cond_wait(&_empty,&lock);
        }
        // 拿出队头的资源
        out = q.front();
        // 删除队头的资源
        q.pop();
        // 唤醒正在等待的生产者
        if(q.size() < _capacity/2)
        {
            pthread_cond_signal(&_full);
            std::cout << "生产者快来生产吧 ";
        }
        // 释放锁
        pthread_mutex_unlock(&lock);
    }

    ~PC()
    {
        _capacity = 0;
        // 销毁锁和条件变量
        pthread_mutex_destroy(&lock);
        pthread_cond_destroy(&_empty);
        pthread_cond_destroy(&_full);
    }

private:
    // 判断队列是否为空
    bool empty()
    {
        return q.empty();
    }
    // 判断队列是否为满
    bool full()
    {
        return q.size() == _capacity;
    }

private:
    std::queue<T> q;
    int _capacity; // 队列的容量
    // 定义锁
    pthread_mutex_t lock;
    // 定义条件变量：_empty表示队列为空的条件变量，_full表示队列为满的条件变量
    pthread_cond_t _empty;
    pthread_cond_t _full;
};
